package top.jolyoulu.力扣._1997_访问完所有房间的第一天;
//你需要访问 n 个房间，房间从 0 到 n - 1 编号。同时，每一天都有一个日期编号，从 0 开始，依天数递增。你每天都会访问一个房间。
//
// 最开始的第 0 天，你访问 0 号房间。给你一个长度为 n 且 下标从 0 开始 的数组 nextVisit 。在接下来的几天中，你访问房间的 次序 将根
//据下面的 规则 决定：
//
//
// 假设某一天，你访问 i 号房间。
// 如果算上本次访问，访问 i 号房间的次数为 奇数 ，那么 第二天 需要访问 nextVisit[i] 所指定的房间，其中 0 <= nextVisit[
//i] <= i 。
// 如果算上本次访问，访问 i 号房间的次数为 偶数 ，那么 第二天 需要访问 (i + 1) mod n 号房间。
//
//
// 请返回你访问完所有房间的第一天的日期编号。题目数据保证总是存在这样的一天。由于答案可能很大，返回对 10⁹ + 7 取余后的结果。
//
//
//
// 示例 1：
//
//
//输入：nextVisit = [0,0]
//输出：2
//解释：
//- 第 0 天，你访问房间 0 。访问 0 号房间的总次数为 1 ，次数为奇数。
//  下一天你需要访问房间的编号是 nextVisit[0] = 0
//- 第 1 天，你访问房间 0 。访问 0 号房间的总次数为 2 ，次数为偶数。
//  下一天你需要访问房间的编号是 (0 + 1) mod 2 = 1
//- 第 2 天，你访问房间 1 。这是你第一次完成访问所有房间的那天。
//
//
// 示例 2：
//
//
//输入：nextVisit = [0,0,2]
//输出：6
//解释：
//你每天访问房间的次序是 [0,0,1,0,0,1,2,...] 。
//第 6 天是你访问完所有房间的第一天。
//
//
// 示例 3：
//
//
//输入：nextVisit = [0,1,2,0]
//输出：6
//解释：
//你每天访问房间的次序是 [0,0,1,1,2,2,3,...] 。
//第 6 天是你访问完所有房间的第一天。
//
//
//
//
// 提示：
//
//
// n == nextVisit.length
// 2 <= n <= 10⁵
// 0 <= nextVisit[i] <= i
//
//
// Related Topics 数组 动态规划 👍 88 👎 0


import top.jolyoulu.utils.TimerUtils;

import java.util.Arrays;
import java.util.Objects;

//leetcode submit region begin(Prohibit modification and deletion)
public class Answer1 {
    /*
        思路：动态规划
        nextVisit 每个元素包含着当前要跳转到的房间下标，每当房间访问次数达到偶数时才能访问下一个房间
        例如 [0,0,0] 每个元素都跳转0房间从头开始访问，那么下标0房间访问2天，才能到底下标1房间
        下标1房间跳转下标0房间，再次从下标0访问到下标1房间，也就是说需要6天才能访问完下标1房间
        第1天 [访问,未访问,未访问]
        第2天 [访问*2,未访问,未访问]
        第3天 [访问*2,访问,未访问]
        第4天 [访问*3,访问,未访问]
        第5天 [访问*4,访问,未访问]
        第6天 [访问*4,访问*2,未访问] 下一天就是访问下标2的房间，所以符合退出条件
    */
    class Solution {

        public int firstDayBeenInAllRooms(int[] nextVisit) {
            int mod = 1000000007;
            int len = nextVisit.length;
            int[] dp = new int[len];

            dp[0] = 2; //初始化原地待一天 + 访问下一个房间一天
            for (int i = 1; i < len; i++) {
                int to = nextVisit[i];
                dp[i] = 2 + dp[i - 1];
                if (to != 0) {
                    dp[i] = (dp[i] - dp[to - 1] + mod) % mod; //避免负数
                }

                dp[i] = (dp[i] + dp[i - 1]) % mod;
            }
            return dp[len - 2]; //题目保证n >= 2

        }
    }

    public static void main(String[] args) {
        Solution solution = new Answer1().new Solution();
        new TimerUtils<int[], Integer>().timer(
                () -> new int[]{0, 0},
                solution::firstDayBeenInAllRooms,
                (res) -> {
                }
        );
        new TimerUtils<int[], Integer>().timer(
                () -> new int[]{0, 0, 2},
                solution::firstDayBeenInAllRooms,
                (res) -> {
                }
        );
        new TimerUtils<int[], Integer>().timer(
                () -> new int[]{0,1,2,0},
                solution::firstDayBeenInAllRooms,
                (res) -> {
                }
        );
        new TimerUtils<int[], Integer>().timer(
                () -> new int[]{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
                solution::firstDayBeenInAllRooms,
                (res) -> {
                }
        );
    }
}
//leetcode submit region end(Prohibit modification and deletion)